import torch
from diffusers import DiffusionPipeline

# 推理配置
# 模型地址如果是单独下载的请在模型目录执行pwd命令查看具体的路径
# 预置路径：model/stable-diffusion-xl-base-1_0
model_path = "model/stable-diffusion-xl-base-1_0"
device="npu"
# 更换提示词重新生成,deepseek联网+深度思考：我正在使用sdxl-base-1.0模型进行文生图创作，需要生成一张中国古风美女的图片，请帮忙写提示词，提示词要小于70个tokens
prompt="Elegant Tang dynasty lady in silk hanfu, phoenix hairpin, porcelain skin. Peach blossoms background, moonlight glow, ink-wash style, ethereal grace --ar 4:3"

# 加载模型
pipe = DiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
pipe.to(device)

# if using torch < 2.0
# pipe.enable_xformers_memory_efficient_attention()

# 推理
image = pipe(prompt=prompt).images[0]
# 保存图片
image.save("output2.png")

# 显示图片
import matplotlib.pyplot as plt
plt.axis("off")
plt.imshow(image)
